import java.util.HashMap;

public class work1 {
    public int subarraysDivByK(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0, ret = 0;
        for (int temp : nums) {
            sum += temp;
            int s = (sum % k + k) % k;
            ret += map.getOrDefault(s, 0);
            map.put(s, map.getOrDefault(s, 0) + 1);
        }
        return ret;
    }
}
